using System.Windows;
using System.Windows.Controls;
using MinoChat.Data.Entities;
namespace MinoChat.Converters;

public class MessageContentTemplateSelector : DataTemplateSelector
{
    //DataTemplate模板需要你自己去实例化
    //渲染纯文本的模板
    public DataTemplate? PlainTextTemplate { get; set; }
    //渲染Markdown的模板
    public DataTemplate? MarkdownTemplate { get; set; }

    public override DataTemplate? SelectTemplate(object item, DependencyObject container)
    {
        if (item is ChatMessage message)
        {
            // AI消息使用Markdown模板，用户消息使用纯文本模板
            return message.IsUserMessage ? PlainTextTemplate : MarkdownTemplate;
        }
        return PlainTextTemplate;
    }

}
